#Anaconda/envs/pyqtenv python
# -*- coding: utf-8 -*-
'''
@File    :   DevMap.py
@Time    :   2024/11/15
@Author  :   CRC1109 
@Version :   1.0
@Desc    :   Built in PyQt5
             
'''

from Netterminal.Sources.Device.DevBase import DevBase
from PyQt5 import QtWebEngineWidgets
from PyQt5.QtCore import QUrl
import os
import json
import urllib
import math

class DevMapHtml(DevBase):
    def __init__(self, name:str, **kwargs):
        """Map device

        Args:
            name (str): 地图控制设备
        """
        super(DevMapHtml,self).__init__(name)
        self.register(**kwargs)
        
        self.X_PI = 3.14159265358979324 * 3000.0 / 180.0
        self.PI = 3.1415926535897932384626
        self.a = 6378245.0 # 长半轴
        self.ee = 0.00669342162296594323 # 偏心率平方

    def register(self, **kwargs):
        self.mapShowWebengine:QtWebEngineWidgets = kwargs.get('mapShowWebengine')

    def ready(self):
        self.mapHtmlDir = os.path.join(os.path.dirname(os.path.abspath(__file__)),"baidumap")
        self.mapHtmlPath = os.path.join(self.mapHtmlDir,"map.html")
        self.mapHtmlQUrl = QUrl(("file:\\" + self.mapHtmlPath).replace('\\', '/'))
        self.loc_init = [126.632628, 45.743215]

        self.mapShowWebengine.page().load(self.mapHtmlQUrl)

    def ctlReload(self):
        """重新加载地图
        """
        pass

    def ctlAddMark(self, Lng:float, Lat:float, popup:str="UAV", icon:str="UAV"):
        """在地图上添加标记点

        Args:
            Lng (float): 经度
            Lat (float): 纬度
            popup (str, optional): 标记的命名. Defaults to "Mark".
            icon (str, optional): 标记的图标. Defaults to "cloud".
        """

        js_command = f"add_point('{popup}','{icon}.png','{str(Lng)}','{str(Lat)}')"
        self.mapShowWebengine.page().runJavaScript(js_command)

    def ctlAddLine(self, Lng1:float, Lat1:float, Lng2:float, Lat2:float, popup:str="Line", color:str="red"):
        """在地图上添加线

        Args:
            Lng1 (float): 起点经度
            Lat1 (float): 起点纬度
            Lng2 (float): 终点经度
            Lat2 (float): 终点纬度
            popup (str, optional): 线的命名. Defaults to "Line".
            color (str, optional): 线的颜色. Defaults to "red".
        """

        js_command = f"add_line('{popup}','{str(Lng1)}','{str(Lat1)}','{str(Lng2)}','{str(Lat2)}','{color}')"
        self.mapShowWebengine.page().runJavaScript(js_command)

    def ctlRemove(self, popup:str="all"):
        """在地图上移除标记点或线

        Args:
            popup (str): 要移除的对象

        """

        if popup == "all":
            js_command = "remove_all()"
        elif popup == "redline":
            js_command = "remove_lineRed()"
        elif js_command == "blackline":
            js_command = "remove_lineBlack()"
        elif js_command == "vector":
            js_command = "remove_vector()"
        elif js_command == "point":
            js_command = "remove_point()"
        else:
            print(self.name + " ctlRemove error with " + popup)
            pass

        self.mapShowWebengine.page().runJavaScript(js_command)

    def bd09_to_gcj02(self, bd_lon, bd_lat):
        """
        百度坐标系(BD-09)转火星坐标系(GCJ-02)
        百度——>谷歌、高德
        :param bd_lat:百度坐标纬度
        :param bd_lon:百度坐标经度
        :return:转换后的坐标列表形式
        """
        x = bd_lon - 0.0065
        y = bd_lat - 0.006
        z = math.sqrt(x * x + y * y) - 0.00002 * math.sin(y * self.X_PI)
        theta = math.atan2(y, x) - 0.000003 * math.cos(x * self.X_PI)
        gg_lng = z * math.cos(theta)
        gg_lat = z * math.sin(theta)
        return [gg_lng, gg_lat]
    
    def gcj02_to_wgs84(self, lng, lat):
        """
        GCJ02(火星坐标系)转GPS84
        :param lng:火星坐标系的经度
        :param lat:火星坐标系纬度
        :return:
        """
        dlat = self._transformlat(lng - 105.0, lat - 35.0)
        dlng = self._transformlng(lng - 105.0, lat - 35.0)
        radlat = lat / 180.0 * self.PI
        magic = math.sin(radlat)
        magic = 1 - self.ee * magic * magic
        sqrtmagic = math.sqrt(magic)
        dlat = (dlat * 180.0) / ((self.a * (1 - self.ee)) / (magic * sqrtmagic) * self.PI)
        dlng = (dlng * 180.0) / (self.a / sqrtmagic * math.cos(radlat) * self.PI)
        mglat = lat + dlat
        mglng = lng + dlng
        return [lng * 2 - mglng, lat * 2 - mglat]
    
    def bd09_to_wgs84(self, bd_lon, bd_lat):
        lon, lat = self.bd09_to_gcj02(bd_lon, bd_lat)
        return self.gcj02_to_wgs84(lon, lat)
    
    def bd09_to_wgs84(self, bd_lon, bd_lat):
        lon, lat = self.bd09_to_gcj02(bd_lon, bd_lat)
        return self.gcj02_to_wgs84(lon, lat)
    
    def gcj02_to_bd09(self, lng, lat):
        """
        火星坐标系(GCJ-02)转百度坐标系(BD-09)
        谷歌、高德——>百度
        :param lng:火星坐标经度
        :param lat:火星坐标纬度
        :return:
        """
        z = math.sqrt(lng * lng + lat * lat) + 0.00002 * math.sin(lat * self.X_PI)
        theta = math.atan2(lat, lng) + 0.000003 * math.cos(lng * self.X_PI)
        bd_lng = z * math.cos(theta) + 0.0065
        bd_lat = z * math.sin(theta) + 0.006
        return [bd_lng, bd_lat]
    
    def wgs84_to_gcj02(self, lng, lat):
        """
        WGS84转GCJ02(火星坐标系)
        :param lng:WGS84坐标系的经度
        :param lat:WGS84坐标系的纬度
        :return:
        """
        dlat = self._transformlat(lng - 105.0, lat - 35.0)
        dlng = self._transformlng(lng - 105.0, lat - 35.0)
        radlat = lat / 180.0 * self.PI
        magic = math.sin(radlat)
        magic = 1 - self.ee * magic * magic
        sqrtmagic = math.sqrt(magic)
        dlat = (dlat * 180.0) / ((self.a * (1 - self.ee)) / (magic * sqrtmagic) * self.PI)
        dlng = (dlng * 180.0) / (self.a / sqrtmagic * math.cos(radlat) * self.PI)
        mglat = lat + dlat
        mglng = lng + dlng
        return [mglng, mglat]
    
    def wgs84_to_bd09(self, lon, lat):
        lon, lat = self.wgs84_to_gcj02(lon, lat)
        return self.gcj02_to_bd09(lon, lat)

    def _transformlng(self, lng, lat):
        ret = 300.0 + lng + 2.0 * lat + 0.1 * lng * lng + \
            0.1 * lng * lat + 0.1 * math.sqrt(math.fabs(lng))
        ret += (20.0 * math.sin(6.0 * lng * self.PI) + 20.0 *
                math.sin(2.0 * lng * self.PI)) * 2.0 / 3.0
        ret += (20.0 * math.sin(lng * self.PI) + 40.0 *
                math.sin(lng / 3.0 * self.PI)) * 2.0 / 3.0
        ret += (150.0 * math.sin(lng / 12.0 * self.PI) + 300.0 *
                math.sin(lng / 30.0 * self.PI)) * 2.0 / 3.0
        return ret
    def _transformlat(self, lng, lat):
        ret = -100.0 + 2.0 * lng + 3.0 * lat + 0.2 * lat * lat + \
            0.1 * lng * lat + 0.2 * math.sqrt(math.fabs(lng))
        ret += (20.0 * math.sin(6.0 * lng * self.PI) + 20.0 *
                math.sin(2.0 * lng * self.PI)) * 2.0 / 3.0
        ret += (20.0 * math.sin(lat * self.PI) + 40.0 *
                math.sin(lat / 3.0 * self.PI)) * 2.0 / 3.0
        ret += (160.0 * math.sin(lat / 12.0 * self.PI) + 320 *
                math.sin(lat * self.PI / 30.0)) * 2.0 / 3.0
        return ret
